package mk.com.ldm.corepoint.implementation.server;

import gwtupload.server.UploadAction;
import gwtupload.server.exceptions.UploadActionException;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Hashtable;
import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.FileItem;

public class UploadServlet extends UploadAction {
	private static final long serialVersionUID = 1L;
	  
		
	
	  Hashtable<String, String> receivedContentTypes = new Hashtable<String, String>();
	  /**
	   * Maintain a list with received files and their content types. 
	   */
	  Hashtable<String, File> receivedFiles = new Hashtable<String, File>();

	  /**
	   * Override executeAction to save the received files in a custom place
	   * and delete this items from session.  
	   */
	  
	  public UploadServlet(){
		  System.out.println("Servlet");
	  }
	  
	  @Override
	  public String executeAction(HttpServletRequest request, List<FileItem> sessionFiles) throws UploadActionException {
	    String response = "";
	    int cont = 0;
	    for (FileItem item : sessionFiles) {
	      if (false == item.isFormField()) {
	        cont ++;
	        try {
	          // Create a new file based on the remote file name in the client
	           String saveName = item.getName().replaceAll("[\\\\/><\\|\\s\"'{}()\\[\\]]+", "_");
	           File file =new File("../src/mk/com/ldm/corepoint/implementation/upload/" + saveName);
	          
	          /// Create a temporary file placed in /tmp (only works in unix)
	           // File file = File.createTempFile("upload-", ".bin", new File("/tmp"));
	          
	          /// Create a temporary file placed in the default system temp folder
//	          File file = File.createTempFile("upload-", ".bin");
//	          System.out.println ("Current dir : " + file.getCanonicalPath());
	          item.write(file);
	          
	          /// Save a list with the received files
	          receivedFiles.put(item.getFieldName(), file);
	          receivedContentTypes.put(item.getFieldName(), item.getContentType());
	          
	          /// Compose a xml message with the full file information which can be parsed in client side
	          response += "<file-" + cont + "-field>" + item.getFieldName() + "</file-" + cont + "-field>\n";
	          response += "<file-" + cont + "-name>" + item.getName() + "</file-" + cont + "-name>\n";
	          response += "<file-" + cont + "-size>" + item.getSize() + "</file-" + cont + "-size>\n";
	          response += "<file-" + cont + "-type>" + item.getContentType()+ "</file-" + cont + "-type>\n";
	          
	         
	        } catch (Exception e) {
	          throw new UploadActionException(e);
	        }
	      }
	    }
	    
	    /// Remove files from session because we have a copy of them
	    removeSessionFileItems(request);
	   
	    /// Send information of the received files to the client.
	    return "<response>\n" + response + "</response>\n";
	  }
	  
	  /**
	   * Get the content of an uploaded file.
	   */
	  @Override
	  public void getUploadedFile(HttpServletRequest request, HttpServletResponse response) throws IOException {
	    String fieldName = request.getParameter(PARAM_SHOW);
	    File f = receivedFiles.get(fieldName);
	    if (f != null) {
	      response.setContentType(receivedContentTypes.get(fieldName));
	      FileInputStream is = new FileInputStream(f);
	      copyFromInputStreamToOutputStream(is, response.getOutputStream());
	    } else {
	      renderXmlResponse(request, response, ERROR_ITEM_NOT_FOUND);
	   }
	  }
	  
	  /**
	   * Remove a file when the user sends a delete request.
	   */
	  @Override
	  public void removeItem(HttpServletRequest request, String fieldName)  throws UploadActionException {
	    File file = receivedFiles.get(fieldName);
	    receivedFiles.remove(fieldName);
	    receivedContentTypes.remove(fieldName);
	    if (file != null) {
	      file.delete();
	    }
	  }
}
